#!/usr/bin/python3
# -*- coding:utf-8 -*-
""""
   @Author: 叶建伟
   @Email: yejw46507@hundsun.com
   @Time: 2022/6/13 15:14
   @File: 可疑交易查询.py
"""
import time

from 名单管理.common import Table, Check
from 线下客户管理.线下客户共用脚本.open_menu import open_menu, arm_form_fill
from 基础公共脚本.步骤检查_hui2 import check_table, check_table_rows,FE

def suspicious_trade_search(input_data=None, ui_expect=None):
    open_menu({'子系统名称': '反洗钱系统', '菜单名称': '可疑交易管理/可疑交易查询'})
    # 资源设置搜索框表单填入
    time.sleep(2)
    search_form = FE.form(idname={"id":"mainSearch"})
    search_form.button_option('重置')
    search_form.auto_fill(input_data)
    search_form.button_option('查询')
    # 检查查询结果
    check_table_rows(input_data=None, ui_expect=ui_expect)

def suspicious_trade_export(input_data=None, ui_expect=None):
    # 进入 风险等级查询 菜单 线下客户管理/线下客户管理
    open_menu({'子系统名称': '反洗钱系统', '菜单名称': '可疑交易管理/可疑交易查询'})
    Table().export_excel(input_data=input_data.get("导出"), ui_expect=None)
    Check().check_export(input_data=None, ui_expect=ui_expect.get("导出"))